package leetcode.binaryTree;

import datastructure.tree.TreeNode;

/**
 * @Description: https://leetcode.cn/problems/balanced-binary-tree/
 * 给定一个二叉树，判断它是否是高度平衡的二叉树。
 *
 * 本题中，一棵高度平衡二叉树定义为：
 *
 * 一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
 *
 * 草稿本：12
 *
 * @Author Ammar
 * @Create 2023/4/15 15:05
 */
public class _110_平衡二叉树 {
    public boolean isBalanced(TreeNode root) {
        return height(root) >= 0;
    }

    // 返回以 node 为根节点的树的高度
    private int height(TreeNode node) {
        if (node == null) return 0;
        int leftHeight = height(node.left);
        if (leftHeight == -1) return -1;
        int rightHeight = height(node.right);
        if (rightHeight == -1) return -1;
        // q: 为什么左右子树绝对值小于2以后，要给其最大值加1？
        // a: 因为我们要返回的是以node为根节点的树的高度，而不是以node.left或node.right为根节点的树的高度
        return Math.abs(leftHeight - rightHeight) < 2 ? Math.max(leftHeight, rightHeight) + 1 : -1;
    }
}
